54cc41
@@ -18,6 +18,7 @@
package org.apache.camel.component.stringtemplate;
 
 import java.io.StringWriter;
 import java.util.Map;
+import java.util.Map.Entry;
 
 import org.antlr.stringtemplate.AutoIndentWriter;
 import org.antlr.stringtemplate.StringTemplate;
@@ -52,9 +53,10 @@
public class StringTemplateEndpoint extends ResourceBasedEndpoint {
     }
 
     @Override
+    @SuppressWarnings("unchecked")
     protected void onExchange(Exchange exchange) throws Exception {
         StringWriter buffer = new StringWriter();
-        Map variableMap = ExchangeHelper.createVariableMap(exchange);
+        Map<String, Object> variableMap = ExchangeHelper.createVariableMap(exchange);
 
         // getResourceAsInputStream also considers the content cache
         String text = exchange.getContext().getTypeConverter().mandatoryConvertTo(String.class, getResourceAsInputStream());
@@ -69,6 +71,9 @@
public class StringTemplateEndpoint extends ResourceBasedEndpoint {
         Message out = exchange.getOut();
         out.setBody(buffer.toString());
         out.setHeader(StringTemplateConstants.STRINGTEMPLATE_RESOURCE, getResource());
+        Map<String, Object> headers = (Map<String, Object>) variableMap.get("headers");
+        for (Entry<String, Object> entry : headers.entrySet()) {
+            out.setHeader(entry.getKey(), entry.getValue());
+        }
     }
-    
 }
\ No newline at end of file
